from config.config import OPENAPI_API_KEY_3, OPENAPI_API_BASE_3
from openai import OpenAI

client = OpenAI(
    api_key=OPENAPI_API_KEY_3,
    base_url=OPENAPI_API_BASE_3
)


def createNews():
    ## 注意此处有多少轮会话都需要传递进去（根据自己的需求）
    messages = [
        {
            "role": "system", "content": f"你是一个小说家，专门负责写小说，根据用户的输入，写一篇作文"
        },
        {
            "role": "user", "content": "小青蛙找妈妈的故事"
        },
        {
            "role": "assistant",
            "content": "### 小青蛙找妈妈的故事\n\n在一个宁静的池塘边，住着一只小青蛙。它的名字叫跳跳，身上穿着一身绿绿的皮肤，眼睛圆圆的，亮晶晶的",
        },
        {
            "role": "user", "content": "继续"
        },
    ]
    response = client.chat.completions.create(
        model="gpt-3.5-turbo-0125",
        messages=messages,
        max_tokens=100
    )
    print(response)
    return {"message": response}
